package com.linrui.basic.service.impl;

import com.linrui.basic.service.IBaiduAiService;
import com.linrui.basic.utils.AjaxResult;
import com.linrui.basic.utils.BaiduAuditUtils;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.Map;

@Service
public class BaiduAiServiceImpl implements IBaiduAiService {
    @Override
    public AjaxResult audit(String text, String imgUrl, String videoUrl) {
        StringBuffer stringBuffer = new StringBuffer();
        // 校验文本
        if(!StringUtils.isEmpty(text)){
            Map<String, Object> map = BaiduAuditUtils.textCensor(text);
            Boolean flag = (Boolean) map.get("success");
            if(!flag){
                stringBuffer.append(text+":"+map.get("message"));
            }
        }

        // 校验图片
        if(!StringUtils.isEmpty(imgUrl)){
            Map<String, Object> map = BaiduAuditUtils.imgCensor(imgUrl);
            Boolean flag = (Boolean) map.get("success");
            if(!flag){
                stringBuffer.append(imgUrl+":"+map.get("message"));
            }
        }

        // 校验视频
        if(!StringUtils.isEmpty(videoUrl)){
            // @TODO
        }
        // 如果说stringBuffer有值,说明有不合规的内容,所以要返回AjaxResult的success=fasle
        if(StringUtils.isEmpty(stringBuffer.toString())){
            return AjaxResult.me();
        }
        return AjaxResult.me().setSuccess(false).setMessage(stringBuffer.toString());
    }
}